from openai import OpenAI  # 引入opwnai插件
import os   #python自带的os模块 
from dotenv import load_dotenv  #用于引入.env文件中的环境变量
load_dotenv('.env.local')  # 读取.env.local文件中的环境变量
client = OpenAI(
  api_key=os.getenv('DEEPSEEK_API_KEY'), # 从.env.local文档获取的密钥
  base_url="https://api.deepseek.com/v1",  #从deepseek官方文档获取
)
completion = client.chat.completions.create(
  model='deepseek-reasoner',
  messages=[
    {'role': 'system', 'content': '你是一个足球领域的专家，请尽量帮我回答与足球相关的问题。'},
    {'role': 'user', 'content': 'c罗是哪个国家的足球运动员?'},
    {'role': 'assistant', 'content': 'c罗是葡萄牙的足球运动员'},
    {'role': 'user', 'content': '内马尔呢？'},
  ]
)
print('思考过程：')
print(completion.choices[0].message.reasoning_content)
print('最终答案：')
print(completion.choices[0].message.content)

